#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>

// 组播地址（选择局域网内可用的组播地址）
#define MULTICAST_GROUP "239.255.1.1"
// 端口号（选择1024-65535之间的未占用端口）
#define PORT 9000
// 发送间隔（秒）
#define SEND_INTERVAL 2
// 缓冲区大小
#define BUFFER_SIZE 1024

int main() {
    int sockfd;
    struct sockaddr_in multicast_addr;
    char buffer[BUFFER_SIZE];
    int count = 0;

    // 创建UDP套接字
    if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
        perror("socket creation failed");
        exit(EXIT_FAILURE);
    }

    // 设置发送接口（指定当前机器的IP地址）
    struct in_addr local_interface;
    // 如果你在电脑1上运行发送端，使用192.168.5.128
    // 如果你在电脑2上运行发送端，使用192.168.5.129
    local_interface.s_addr = inet_addr("192.168.5.128");
    if (setsockopt(sockfd, IPPROTO_IP, IP_MULTICAST_IF, 
                  &local_interface, sizeof(local_interface)) < 0) {
        perror("setsockopt IP_MULTICAST_IF failed");
        exit(EXIT_FAILURE);
    }

    // 初始化组播地址结构
    memset(&multicast_addr, 0, sizeof(multicast_addr));
    multicast_addr.sin_family = AF_INET;
    multicast_addr.sin_addr.s_addr = inet_addr(MULTICAST_GROUP);
    multicast_addr.sin_port = htons(PORT);

    printf("Multicast sender started. Sending to %s:%d\n", 
           MULTICAST_GROUP, PORT);
    printf("Press Ctrl+C to stop...\n");

    // 循环发送数据
    while (1) {
        // 准备发送的数据
        snprintf(buffer, BUFFER_SIZE, "Message %d from %s", 
                count++, inet_ntoa(local_interface));
        
        // 发送数据到组播地址
        ssize_t send_len = sendto(sockfd, buffer, strlen(buffer), 0,
                                 (struct sockaddr*)&multicast_addr, 
                                 sizeof(multicast_addr));
        
        if (send_len < 0) {
            perror("sendto failed");
            exit(EXIT_FAILURE);
        }

        printf("Sent: %s\n", buffer);
        sleep(SEND_INTERVAL);
    }

    close(sockfd);
    return 0;
}
